#include <memory>
#include "UdpServer.hpp"
#include "Route.hpp"
#include "ThreadPool.hpp"

using namespace ThreadPoolModule;

using task_t = std::function<void()>;

// ./udpserver port
int main(int argc, char* argv[])
{
    if(argc != 2)
    {
        std::cerr << "Usage: " << argv[0] << " port" << std::endl;
        return 1;
    }

    uint16_t port = std::stoi(argv[1]);

    Enable_Console_Log_Strategy();

    // 路由服务
    Route route;

    // 线程池
    auto tp = ThreadPool<task_t>::GetInstance();

    // 网络服务器对象提供网络通信功能
    std::unique_ptr<UdpServer> usvr = std::make_unique<UdpServer>(port, [&route, &tp](int sockfd, const std::string& message, InetAddr& client){
        task_t t = std::bind(&Route::MessageRoute, &route, sockfd, message, client);
        tp->Enqueue(t);
    });
    usvr->Init();
    usvr->Start();
    return 0;
}
